package com.bee.剑指offer;

import java.util.ArrayList;

/**
 * 小明很喜欢数学,有一天他在做数学作业时,要求计算出9~16的和,他马上就写出了正确答案是100。
 * 但是他并不满足于此,他在想究竟有多少种连续的正数序列的和为100(至少包括两个数)。
 * 没多久,他就得到另一组连续正数和为100的序列:18,19,20,21,22。
 * 现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列?
 */
public class 和为S的连续正数序列 {

    /**
     * 双指针算法
     */
    public static ArrayList<ArrayList<Integer>> FindContinuousSequence(int sum) {
        ArrayList<ArrayList<Integer>> result = new ArrayList<>();
        if (sum <= 2) {
            return result;
        }
        int low = 1;
        int high = 2;
        while (high > low) {
            // 等差数列，两数之和公式
            int cur = ((high + low) * (high - low + 1)) / 2;
            if (cur == sum) {
                ArrayList<Integer> list = new ArrayList<>();
                for (int i = low; i <= high; i++) {
                    list.add(i);
                }
                low++;
                result.add(list);
            }
            if (cur < sum) {
                high++;
            }
            if (cur > sum) {
                low++;
            }
        }
        return result;
    }

    public static void main(String[] args) {
        ArrayList<ArrayList<Integer>> arrayLists = FindContinuousSequence(9);
        System.out.println(arrayLists.toString());
    }
}
